#-------------------------------------------------------------
#
# (C) Copyright IBM Corp. 2010, 2015
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#-------------------------------------------------------------


args <- commandArgs(TRUE)
options(digits=22)

library("Matrix")

V1 <- readMM(paste(args[1], "V.mtx", sep=""))
V <- as.matrix(V1);
n <- ncol(V); 

R1 <- array(0,dim=c(1,n))
R2 <- array(0,dim=c(1,n))

for( i in 1:n )
{
   X <- V[ ,i];                 
   R1[1,i] <- sum(X);
}   

if( args[3]==1 )
{  
  for( i in 1:n )
  {
     X1 <- V[i,]; 
     X2 <- V[i,];                 
     R2[1,i] <- R1[1,i] + sum(X1)+sum(X2);
  }   
} else {
  for( i in 1:n )
  {
     X1 <- V[i,]; 
     X2 <- V[,i];                 
     R2[1,i] <- R1[1,i] + sum(X1)+sum(X2);
  }  
}

writeMM(as(R2, "CsparseMatrix"), paste(args[2], "Rout", sep="")); 